// udp_server.hpp
#pragma once
#include <sys/types.h>
#include <sys/socket.h>
#include <cstring>
#include <cerrno>
#include <iostream>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string>
#include <functional>
using namespace std;

enum
{
    USAGE_ERR,  // 运行服务器时输入错误
    SOCKET_ERR, // 创建套接字失败
    BIND_ERR,   // 绑定失败
};

using func_t = function<string(const string &)>; // 和typedef一样的功能

// 127.0.0.1: 本地环回，就表示的就是当前主机，通常用来进行本地通信或者测试
const static int static_port = 8080;

class Server
{
public:
    // 构造初始化
    Server(func_t service, int port = static_port)
        : _func(service),
          _port(port)
    {
        _func = service;
        _port = port;
    }

    void Init()
    {

        // 1.创建socket接口，打开网络文件
        // AF_INET表示是IPv4，sock_DGRAM表示无连接服务，UDP使用
        _socket = socket(AF_INET, SOCK_DGRAM, 0);
        if (_socket < 0)
        {
            cout << strerror(errno) << endl;
            exit(SOCKET_ERR);
        }
        cout << "create socket success:" << _socket << endl;

        // 2.给服务器指定IP地址和port
        struct sockaddr_in local; // 这个local在栈帧上，不在内核中，没有和内核建立联系
        bzero(&local, sizeof(local));
        local.sin_port = htons(_port); // 端口
        // 云服务器，或者一款服务器，一般不要指明某一个确定的ip,需要让服务器自己指定IP地址，自己本地安装的虚拟机或者物理机器允许
        local.sin_addr.s_addr = INADDR_ANY;
        local.sin_family = AF_INET;
        if (bind(_socket, (struct sockaddr *)&local, sizeof(local)) < 0)
        {
            cout << strerror(errno) << endl;
            exit(BIND_ERR);
        }
        cout << "bind socket success" << endl;
        // cout<<local.sin_addr.s_addr<<":"<<local.sin_port<<endl;
    }

    void Start()
    {
        // 服务器正式运行
        while (true)
        {
            char buff[1024];
            struct sockaddr_in client;
            socklen_t len = sizeof(client);
            // 1.先接受客户端的命令
            int n = recvfrom(_socket, buff, sizeof(buff) - 1, 0, (struct sockaddr *)&client, &len);
            if (n > 0)
            {
                buff[n] = '\0';
            }
            else
            {
                continue;
            }

            // 2.提取客户端信息-debug
            string clientip = inet_ntoa(client.sin_addr);
            int clientport = ntohs(client.sin_port);
            cout << clientip << ":" << clientport << "#" << buff << endl;

            // 3.进行业务处理
            string result = _func(buff);

            // debug--通过测试，可以发现如果命令输入错误，result的大小为0，是一个空字符串
            // printf("result:%d\n",result[0]);
            // cout<<"size"<<result.size()<<endl;

            // 4.将处理结果发给客户端
            sendto(_socket, result.c_str(), result.size(), 0, (struct sockaddr *)&client, len);
        }
    }

    ~Server()
    {
    }

private:
    int _socket;  // 套接字描述符
    int _port;    // 端口
    func_t _func; // 处理业务接口
};
